# -*- coding: utf-8 -*-
import sys

sys.path.append('../')
from common.config import *
from db.mysql import MysqlClient
from itertools import product
import requests
from common.helper import Helper
from urllib.parse import  urlencode


class Spider(object):

    def __init__(self):

        self.mysql = MysqlClient()
        self.name = "泰康健康保保险产品计划"
    # 生产产品数据
    def genProductInfo(self):
        plan = {
            #年龄段
            "year":[str(i) for i in range(0,51)],
            #性别
            "gender":["M","F"],
            #保额
            "money":["50000","100000","150000","200000","300000","400000","500000"],
            #保障期限
            "expire":["30年"],
            #缴费间隔
            "interval":["10"],
            #缴费方式
            "payType":["年缴","月缴"],
            #投保对象
            "payObject":["本人/配偶/儿女/父母"]
        }
        for v in product(plan["year"],plan["gender"],plan["money"],plan["expire"],plan["interval"],plan["payType"],plan["payObject"]):
            self.mysql.cursor.execute( domain[self.name]["insert_sql"].format(
                v[0],v[1],v[2],v[3],v[4],v[5],v[6]
            ))
            self.mysql._conn.commit()
            print( domain[self.name]["insert_sql"].format(
                v[0],v[1],v[2],v[3],v[4],v[5],v[6]
            ))
    #执行爬虫
    def requestJson(self,dt=None):
        data = {

            "codeFlag": "2",
            "productNo": "B29",
            "poltype": "01",
            "responsibility": "B29A",
            "insuredGender": dt["gender"],
            "birthday": "{year}-01-01".format(year = str(2020-int(dt["year"]))),
            "applicantGender": "F",
            "holderBirthday": "{year}-01-01".format(year = str(2020-int(dt["year"]))),
            "paymentDuration": "10",
            "paymentMode": "01" if dt["payType"]=="年缴" else "12",
            "insureDuration": "30",
            "coverageAmount": str(int(dt["money"])/10000)
        }
        payload = urlencode(data).replace("+","")
        r = requests.post( url = domain[self.name]["url"],headers=domain[self.name]["header"],data=payload)
        if r.status_code == 200:
            price = Helper.decimalParse(r.json()["grossFirstPremium"])
            data = {
                "name":self.name,
                "id":dt["id"],
                "price":price
            }
            self.mysql.updatePrice( data )
        else:
            data={
                "name":self.name,
                "id":dt["id"],
                "state":3
            }
            self.mysql.updateTask( data )
            print("\n/*更新产品数据价格失败:%d*/"%dt["id"])

    def main(self):

        while True:

            # 随机领取一条数据
            dt = self.mysql.randTask(self.name)

            if not dt:
                print("产品价格抓取完成~~~")

                return

                # 任务执行状态
            data = {
                "name": self.name,
                "id": dt["id"],
                "state": 1
            }

            self.mysql.updateTask(data)

            self.requestJson(dt)

